/*

 * Licensed to the Apache Software Foundation (ASF) under one

 * or more contributor license agreements.  See the NOTICE file

 * distributed with this work for additional information

 * regarding copyright ownership.  The ASF licenses this file

 * to you under the Apache License, Version 2.0 (the

 * "License"); you may not use this file except in compliance

 * with the License.  You may obtain a copy of the License at

 *

 *     http://www.apache.org/licenses/LICENSE-2.0

 *

 * Unless required by applicable law or agreed to in writing, software

 * distributed under the License is distributed on an "AS IS" BASIS,

 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.

 * See the License for the specific language governing permissions and

 * limitations under the License.

 */

package com.bff.gaia.unified.runners.core.construction;



import com.google.auto.service.AutoService;

import com.bff.gaia.unified.model.pipeline.v1.RunnerApi;

import com.bff.gaia.unified.model.pipeline.v1.RunnerApi.FunctionSpec;

import com.bff.gaia.unified.sdk.runners.AppliedPTransform;

import com.bff.gaia.unified.sdk.transforms.Impulse;

import com.bff.gaia.unified.sdk.transforms.PTransform;



import java.io.IOException;

import java.util.Collections;

import java.util.Map;



/**

 * Utility methods for translating a {@link Impulse} to and from {@link RunnerApi} representations.

 */

public class ImpulseTranslation {

  private static class ImpulseTranslator implements PTransformTranslation.TransformPayloadTranslator<Impulse> {

    @Override

    public String getUrn(Impulse transform) {

      return PTransformTranslation.IMPULSE_TRANSFORM_URN;

    }



    @Override

    public FunctionSpec translate(

        AppliedPTransform<?, ?, Impulse> application, SdkComponents components) throws IOException {

      return FunctionSpec.newBuilder().setUrn(getUrn(application.getTransform())).build();

    }

  }



  /** Registers {@link ImpulseTranslator}. */

  @AutoService(TransformPayloadTranslatorRegistrar.class)

  public static class Registrar implements TransformPayloadTranslatorRegistrar {

    @Override

    public Map<? extends Class<? extends PTransform>, ? extends PTransformTranslation.TransformPayloadTranslator>

        getTransformPayloadTranslators() {

      return Collections.singletonMap(Impulse.class, new ImpulseTranslator());

    }

  }

}